#!/usr/bin/python
# -*- coding: utf8 -*-
import re,os,glob
import datetime
import colorama
from colorama import Fore, Style

# 初始化 colorama
colorama.init(autoreset=True)
#解决快平台print颜色问题

def printx(color,message):
    print(f"{color} {message}")


#获取路径下所有指定后缀的文件名称
def get_files_with_suffix(directory, suffix):
    """
    # 使用 glob 模块搜索指定目录下所有符合模式的文件
    :param directory: 路径
    :param suffixsuffix: 文件后缀
    """
    search_path = os.path.join(directory, f'*{suffix}')
    files = glob.glob(search_path)
    return [os.path.basename(f) for f in files]


def check_recod_path(folder_path):
    if os.path.exists('data/'+folder_path) and os.path.isdir('data/'+folder_path):
        if  os.listdir('data/'+folder_path):
            return folder_path
        else:
            exit(f"❌ {folder_path}文件夹为空，退出程序")
    else:
        exit(f"❌ {folder_path}文件夹不存在，退出程序")




